#!/usr/local/bin/lua
--[[
  udreportcnv.lua - UltraDefrag report converter.
  Converts lua reports to HTML and other formats.
  Copyright (c) 2008 by Dmitri Arkhangelski (dmitriar@gmail.com).

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.
 
  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.
 
  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
--]]

-- USAGE: lua udreportcnv.lua <luar file with full path> <path to Windows directory> [-v]

-- parse command line
assert(arg[1],"Lua Report file name must be specified!")
assert(arg[2],"Path to the Windows directory\nmust be specified as second parameter!")

-- read options
dofile(arg[2] .. "\\UltraDefrag\\options\\udreportopts.lua")

-- source file reading
dofile(arg[1])


-- common functions

function write_ansi(f, ...)
	for i, v in ipairs(arg) do f:write(v) end
end

function write_unicode(f, ...)
	local b, j
	for i, v in ipairs(arg) do
		j = 1
		while true do
			b = string.byte(v, j)
			if b == nil then break end
			f:write(string.char(b), "\0")
			j = j + 1
		end
	end
end

-- write filename to html table; maximum 50 characters per line
function write_ansi_name(f, name)
	local i, j, k, n, len
	local N = max_chars_per_line
	len = string.len(name)
	i = 1
	while len > 0 do
		if len <= N then f:write(string.sub(name,i,i + len - 1)) ; break end 
		j = string.find(name,"\\",i,true)
		if j == nil or j > (i + N) then
			n = N
		else
			k = j
			while true do
				k = string.find(name,"\\",k + 1,true)
				if k == nil then break end
				if k <= (i + N) then j = k else break end
			end
			n = j - i + 1
		end
		f:write(string.sub(name,i,i + n - 1), "<br />")
		if n == 0 then break end -- erroneous situation
		i = i + n
		len = len - n
	end
end

function write_chars(f, chars, i, j)
	local k
	for k=i,j do f:write(chars[k]) end
end

function find_char(chars, ch, i)
	local k = i
	local result = nil
	while chars[k] ~= nil do
		if chars[k] == ch and chars[k + 1] == '\0' then
			result = k ; break
		end
		k = k + 1
	end
	return result
end

function write_unicode_name(f, chars, len)
	local i, j, k, n
	local N = max_chars_per_line * 2
	i = 1
	while len > 0 do
		if len <= N then write_chars(f,chars,i,i + len - 1) ; break end 
		j = find_char(chars,'\\',i)
		if j == nil or j > (i + N) then
			n = N
		else
			k = j
			while true do
				k = find_char(chars,'\\',k + 1)
				if k == nil then break end
				if k <= (i + N) then j = k else break end
			end
			n = j - i + 1 + 1 -- because unicode backslash is '\\\0'
		end
		write_chars(f,chars,i,i + n - 1)
		write_unicode(f,"<br />")
		if n == 0 then break end -- erroneous situation
		i = i + n
		len = len - n
	end
end

-- converters

table_head = [[
<tr>
<td class="c"><a href="javascript:sort_items('fragments')" style="color: #0000FF"># fragments</a></td>
<td class="c"><a href="javascript:sort_items('name')" style="color: #0000FF">filename</a></td>
<td class="c"><a href="javascript:sort_items('comment')" style="color: #0000FF">comment</a></td>
</tr>
]]

end_of_page = [[
</center>
<script type="text/javascript">
init_sorting_engine();
</script>
</body></html>
]]

links_x1 = [[
<table width="100%"><tbody>
<tr>
<td style="text-align: left"><a href="http://ultradefrag.sourceforge.net" style="color: #0000FF">Visit our Homepage</a></td>
<td style="text-align: center"><a 
]]

links_x2 = [[
style="color: #0000FF">View report options</a></td>
<td style="text-align: right">
<a href="http://www.lua.org/" style="color: #0000FF">Powered by Lua</a>
</td>
</tr>
</tbody></table>
]]

function produce_html_output()
	local filename
	local pos = 0
	local js
	local links = links_x1 .. "href=\"file:///" .. arg[2]
	links = links .. "\\UltraDefrag\\options\\udreportopts.lua\" " .. links_x2

	repeat
		pos = string.find(arg[1],"\\",pos + 1,true)
		if pos == nil then filename = "fraglist.htm" ; break end
	until string.find(arg[1],"\\",pos + 1,true) == nil
	filename = string.sub(arg[1],1,pos) .. "fraglist.htm"

	-- note that 'b' flag is needed for utf-16 files
	local f = assert(io.open(filename,"wb"))

	local write_data
	if use_utf16 == 0 then
		write_data = write_ansi
	else
		write_data = write_unicode
	end

	if(enable_sorting == 1) then
		-- read udsorting.js file contents
		local f2 = assert(io.open(arg[2] .. "\\UltraDefrag\\scripts\\udsorting.js", "r"))
	    js = f2:read("*all")
	    f2:close()
	else
		js = "function init_sorting_engine(){}\nfunction sort_items(criteria){}\n"
	end
	write_data(f,
		"<html><head><title>Fragmented files on ", volume_letter,
		":</title>\n", style,
		"<script language=\"javascript\">\n", js,
		"</script>\n</head>\n",
		"<body>\n<center>\n", title_tags.open,
		"Fragmented files on ", volume_letter,
		":", title_tags.close,
		"\n", links, "\n",
		"<div id=\"for_msie\">\n",
		"<table id=\"main_table\" ", table_style, ">\n",
		table_head
		)
	for i, v in ipairs(files) do
		local class
		if v.filtered == 1 then class = "f" else class = "u" end
		write_data(f,
			"<tr class=\"", class, "\"><td class=\"c\">", v.fragments,
			"</td><td>"
			)
		if use_utf16 == 0 then
			-- each <> brackets must be replaced with square brackets
			local tmp = string.gsub(v.name,"<","[")
			tmp = string.gsub(tmp,">","]")
			if split_long_names == 1 then
				write_ansi_name(f,tmp)
			else
				f:write(tmp)
			end
		else
			if split_long_names == 1 then
				local chars = {}
				local i = 1
				for j, b in ipairs(v.uname) do
					chars[i] = string.char(b)
					i = i + 1
				end
				write_unicode_name(f,chars,i - 1)
			else
				for j, b in ipairs(v.uname) do
					f:write(string.char(b))
				end
			end
		end
		write_data(f,
			"</td><td class=\"c\">", v.comment,
			"</td></tr>\n"
			)
	end
	write_data(f,
		--table_head,
		"</table>\n</div>\n", links, "\n"
		)
	write_data(f,end_of_page)
	f:close()
	
	-- if option -v is specified, open report in default web browser
	if arg[3] == "-v" then
		if os.shellexec ~= nil then
			os.shellexec(filename,"open")
		else
			os.execute("cmd.exe /C " .. filename)
		end
	end
end


-- main source code
-- convertion to other (readable) formats
if produce_html == 1 then
	produce_html_output()
end